package com.smxy.mall.utils;

import com.smxy.mall.common.CustomException;
import com.smxy.mall.common.Response;
import com.smxy.mall.entity.User;
import com.smxy.mall.model.Audience;
import com.smxy.mall.model.Current;
import io.jsonwebtoken.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.crypto.spec.SecretKeySpec;
import javax.xml.bind.DatatypeConverter;
import java.security.Key;
import java.util.Date;

/**
 * jwt 工具类
 */
public class JwtTokenUtil {

    //日志
    private static Logger log = LoggerFactory.getLogger(JwtTokenUtil.class);

    public static final String AUTH_HEADER_KEY = "token";


    /**
     * 解析jwt
     * @param jsonWebToken
     * @param base64Security
     * @return
     */
    public static Claims parseJWT(String jsonWebToken, String base64Security) {
        try {
            Claims claims = Jwts.parser()
                    .setSigningKey(DatatypeConverter.parseBase64Binary(base64Security))
                    .parseClaimsJws(jsonWebToken).getBody();
            return claims;
        } catch (ExpiredJwtException  eje) {
            log.error("===== Token过期 =====", eje);
            throw new CustomException(Response.fail("401","Token过期"));
        } catch (Exception e){
            log.error("===== token解析异常 =====", e);
            throw new CustomException(Response.fail("401","token解析异常"));
        }
    }

    /**
     * 构建jwt
     * @param user
     * @param audience jwt配置
     * @return
     */
    public static String createJWT(User user, Audience audience) {
        try {
            // 使用HS256加密算法
            SignatureAlgorithm signatureAlgorithm = SignatureAlgorithm.HS256;

            long nowMillis = System.currentTimeMillis();
            Date now = new Date(nowMillis);

            //生成签名密钥
            byte[] apiKeySecretBytes = DatatypeConverter.parseBase64Binary(audience.getBase64Secret());
            Key signingKey = new SecretKeySpec(apiKeySecretBytes, signatureAlgorithm.getJcaName());

            //userId是重要信息，进行加密下
//            String encryId = Base64Util.encode(userId);

            //添加构成JWT的参数
            JwtBuilder builder = Jwts.builder().setHeaderParam("typ", "JWT")
                    // 可以将基本不重要的对象信息放到claims
                    .claim("type", user.getType())
                    .claim("userId", user.getId())
                    .claim("phone",user.getPhone())
                    .claim("userName",user.getUserName())
                    .claim("openId",user.getOpenId())
                    .setSubject(String.valueOf(user.getId()))           // 代表这个JWT的主体，即它的所有人
                    .setIssuer(audience.getClientId())              // 代表这个JWT的签发主体；
                    .setIssuedAt(new Date())        // 是一个时间戳，代表这个JWT的签发时间；
                    .setAudience(audience.getName())          // 代表这个JWT的接收对象；
                    .signWith(signatureAlgorithm, signingKey);
            //添加Token过期时间
            int TTLMillis = audience.getExpiresSecond();
            if (TTLMillis >= 0) {
                long expMillis = nowMillis + TTLMillis;
                Date exp = new Date(expMillis);
                builder.setExpiration(exp)  // 是一个时间戳，代表这个JWT的过期时间；
                        .setNotBefore(now); // 是一个时间戳，代表这个JWT生效的开始时间，意味着在这个时间之前验证JWT是会失败的
            }

            //生成JWT
            return builder.compact();
        } catch (Exception e) {
            log.error("签名失败", e);
            throw new CustomException(Response.fail("401","签名失败"));
        }
    }

    /**
     * 从token中获取uerId (Subject)
     * @param token
     * @param base64Security
     * @return
     */
    public static Integer getUserId(String token, String base64Security){
        return Integer.parseInt(parseJWT(token, base64Security).getSubject());
    }

    /**
     * 解析token，获取token中的用户信息
     * @param token
     * @param base64Security
     * @return
     */
    public static Current getCurrentUser(String token, String base64Security){
        Claims claims = parseJWT(token, base64Security);
        Current currentCurrent = new Current();
        currentCurrent.setUserId(Integer.parseInt(claims.getSubject()));
        currentCurrent.setUserName(claims.get("userName",String.class));
        currentCurrent.setPhone(claims.get("phone",String.class));
        currentCurrent.setType(claims.get("type",String.class));
        currentCurrent.setIssuer(claims.getIssuer());
        currentCurrent.setAudience(claims.getAudience());
        currentCurrent.setIssuedAt(claims.getIssuedAt());
        return currentCurrent;
    }

    /**
     * 是否已过期
     * @param token
     * @param base64Security
     * @return
     */
    public static boolean isExpiration(String token, String base64Security) {
        return parseJWT(token, base64Security).getExpiration().before(new Date());
    }
}
